#!/usr/bin/env python3
# coding=utf-8
# 导入flask的脚本控制模块
import datetime

from flask_script import Manager
# 导入Flask脚本模块
from flask_migrate import Migrate, MigrateCommand

# 导入工厂函数,sqlalchemy实例,数据库模型模块
from info import create_app, db, models
from info.models import User

app = create_app("development")

# 实例化管理器对象
manage = Manager(app)
# 使用迁移框架
Migrate(app,db)
# 添加迁移命令
manage.add_command('db', MigrateCommand)

def add_test_users():
    """
    添加测试用户
    :return:
    """
    users = []
    now = datetime.datetime.now()
    for num in range(0, 10000):
        try:
            user = User()
            user.nick_name = "%011d" % num
            user.mobile = "%011d" % num
            user.password_hash = "pbkdf2:sha256:50000$SgZPAbEj$a253b9220b7a916e03bf27119d401c48ff4a1c81d7e00644e0aaf6f3a8c55829"
            user.last_login = now - datetime.timedelta(seconds=random.randint(0, 2678400))
            users.append(user)
            print(user.mobile)
        except Exception as e:
            print(e)
    with app.app_context():
        db.session.add_all(users)
        db.session.commit()
    print('OK')




if __name__ == '__main__':
    # print(app.url_map)
    # add_test_users()
    app.run()
    # manage.run()





